package org.basis.netty.practice.protocol;

import io.netty.buffer.ByteBuf;
import io.netty.channel.ChannelHandlerContext;
import io.netty.handler.codec.ByteToMessageCodec;
import lombok.extern.slf4j.Slf4j;
import org.basis.netty.chat.message.Message;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.ObjectInputStream;
import java.io.ObjectOutputStream;
import java.util.List;

/**
 * 自定义消息编码解码协议
 *
 * @author Mr_wenpan@163.com 2021/09/22 09:56
 * @ChannelHandler.Sharable 表示是线程安全的（即一个对象可以被多个线程访问）
 */
@Slf4j
//@ChannelHandler.Sharable
public class MessageCodec extends ByteToMessageCodec<Message> {

    /**
     * 消息编码，在消息出站的时候对消息按照自定义的协议进行编码
     *
     * @param ctx 上下文
     * @param msg 待出站的消息
     * @param out byteBuf
     * @author Mr_wenpan@163.com 2021/9/22 10:11 上午
     */
    @Override
    public void encode(ChannelHandlerContext ctx, Message msg, ByteBuf out) throws Exception {

        // 1. 4个直接的魔术
        out.writeBytes(new byte[]{1, 2, 3, 4});
        // 2. 1个字节的版本号
        out.writeByte(1);
        // 3. 1个字节的序列化方式： 0 ： jdk ， 1 ： JSON
        out.writeByte(0);
        // 4. 1个字节的指令类型
        out.writeByte(msg.getMessageType());
        // 5. 4个字节的序列号
        out.writeInt(msg.getSequenceId());
        // 1个字节对齐填充（无意义）
        out.writeByte(0xff);
        // 6. 获取内容的字节数组
        ByteArrayOutputStream bos = new ByteArrayOutputStream();
        ObjectOutputStream oos = new ObjectOutputStream(bos);
        oos.writeObject(msg);
        byte[] bytes = bos.toByteArray();
        // 7. 4个字节消息长度
        out.writeInt(bytes.length);
        // 8. 写入内容到byteBuf中
        out.writeBytes(bytes);

    }

    /**
     * 消息入站时，按照自定义协议对消息进行解码
     *
     * @param ctx 上下文
     * @param in  ByteBuf
     * @param out 消息
     * @author Mr_wenpan@163.com 2021/9/22 10:12 上午
     */
    @Override
    public void decode(ChannelHandlerContext ctx, ByteBuf in, List<Object> out) throws Exception {

        int magicNum = in.readInt();
        byte version = in.readByte();
        byte serializerType = in.readByte();
        byte messageType = in.readByte();
        int sequenceId = in.readInt();
        in.readByte();
        int length = in.readInt();
        // 消息正文数组
        byte[] bytes = new byte[length];

        // 消息读取到bytes中
        in.readBytes(bytes, 0, length);
        ObjectInputStream ois = new ObjectInputStream(new ByteArrayInputStream(bytes));
        // 将输入流转换为消息对象
        Message message = (Message) ois.readObject();
        log.debug("{}, {}, {}, {}, {}, {}", magicNum, version, serializerType, messageType, sequenceId, length);
        log.debug("{}", message);
        out.add(message);

    }

}
